# -*- coding: utf-8 -*-

import os,math

from settings import *



colormap = {
    # X11 colour table (from "CSS3 module: Color working draft"), with
    # gray/grey spelling issues fixed.  This is a superset of HTML 4.0
    # colour names used in CSS 1.
    "aliceblue": "#f0f8ff",
    "antiquewhite": "#faebd7",
    "aqua": "#00ffff",
    "aquamarine": "#7fffd4",
    "azure": "#f0ffff",
    "beige": "#f5f5dc",
    "bisque": "#ffe4c4",
    "black": "#000000",
    "blanchedalmond": "#ffebcd",
    "blue": "#0000ff",
    "blueviolet": "#8a2be2",
    "brown": "#a52a2a",
    "burlywood": "#deb887",
    "cadetblue": "#5f9ea0",
    "chartreuse": "#7fff00",
    "chocolate": "#d2691e",
    "coral": "#ff7f50",
    "cornflowerblue": "#6495ed",
    "cornsilk": "#fff8dc",
    "crimson": "#dc143c",
    "cyan": "#00ffff",
    "darkblue": "#00008b",
    "darkcyan": "#008b8b",
    "darkgoldenrod": "#b8860b",
    "darkgray": "#a9a9a9",
    "darkgrey": "#a9a9a9",
    "darkgreen": "#006400",
    "darkkhaki": "#bdb76b",
    "darkmagenta": "#8b008b",
    "darkolivegreen": "#556b2f",
    "darkorange": "#ff8c00",
    "darkorchid": "#9932cc",
    "darkred": "#8b0000",
    "darksalmon": "#e9967a",
    "darkseagreen": "#8fbc8f",
    "darkslateblue": "#483d8b",
    "darkslategray": "#2f4f4f",
    "darkslategrey": "#2f4f4f",
    "darkturquoise": "#00ced1",
    "darkviolet": "#9400d3",
    "deeppink": "#ff1493",
    "deepskyblue": "#00bfff",
    "dimgray": "#696969",
    "dimgrey": "#696969",
    "dodgerblue": "#1e90ff",
    "firebrick": "#b22222",
    "floralwhite": "#fffaf0",
    "forestgreen": "#228b22",
    "fuchsia": "#ff00ff",
    "gainsboro": "#dcdcdc",
    "ghostwhite": "#f8f8ff",
    "gold": "#ffd700",
    "goldenrod": "#daa520",
    "gray": "#808080",
    "grey": "#808080",
    "green": "#008000",
    "greenyellow": "#adff2f",
    "honeydew": "#f0fff0",
    "hotpink": "#ff69b4",
    "indianred": "#cd5c5c",
    "indigo": "#4b0082",
    "ivory": "#fffff0",
    "khaki": "#f0e68c",
    "lavender": "#e6e6fa",
    "lavenderblush": "#fff0f5",
    "lawngreen": "#7cfc00",
    "lemonchiffon": "#fffacd",
    "lightblue": "#add8e6",
    "lightcoral": "#f08080",
    "lightcyan": "#e0ffff",
    "lightgoldenrodyellow": "#fafad2",
    "lightgreen": "#90ee90",
    "lightgray": "#d3d3d3",
    "lightgrey": "#d3d3d3",
    "lightpink": "#ffb6c1",
    "lightsalmon": "#ffa07a",
    "lightseagreen": "#20b2aa",
    "lightskyblue": "#87cefa",
    "lightslategray": "#778899",
    "lightslategrey": "#778899",
    "lightsteelblue": "#b0c4de",
    "lightyellow": "#ffffe0",
    "lime": "#00ff00",
    "limegreen": "#32cd32",
    "linen": "#faf0e6",
    "magenta": "#ff00ff",
    "maroon": "#800000",
    "mediumaquamarine": "#66cdaa",
    "mediumblue": "#0000cd",
    "mediumorchid": "#ba55d3",
    "mediumpurple": "#9370db",
    "mediumseagreen": "#3cb371",
    "mediumslateblue": "#7b68ee",
    "mediumspringgreen": "#00fa9a",
    "mediumturquoise": "#48d1cc",
    "mediumvioletred": "#c71585",
    "midnightblue": "#191970",
    "mintcream": "#f5fffa",
    "mistyrose": "#ffe4e1",
    "moccasin": "#ffe4b5",
    "navajowhite": "#ffdead",
    "navy": "#000080",
    "oldlace": "#fdf5e6",
    "olive": "#808000",
    "olivedrab": "#6b8e23",
    "orange": "#ffa500",
    "orangered": "#ff4500",
    "orchid": "#da70d6",
    "palegoldenrod": "#eee8aa",
    "palegreen": "#98fb98",
    "paleturquoise": "#afeeee",
    "palevioletred": "#db7093",
    "papayawhip": "#ffefd5",
    "peachpuff": "#ffdab9",
    "peru": "#cd853f",
    "pink": "#ffc0cb",
    "plum": "#dda0dd",
    "powderblue": "#b0e0e6",
    "purple": "#800080",
    "red": "#ff0000",
    "rosybrown": "#bc8f8f",
    "royalblue": "#4169e1",
    "saddlebrown": "#8b4513",
    "salmon": "#fa8072",
    "sandybrown": "#f4a460",
    "seagreen": "#2e8b57",
    "seashell": "#fff5ee",
    "sienna": "#a0522d",
    "silver": "#c0c0c0",
    "skyblue": "#87ceeb",
    "slateblue": "#6a5acd",
    "slategray": "#708090",
    "slategrey": "#708090",
    "snow": "#fffafa",
    "springgreen": "#00ff7f",
    "steelblue": "#4682b4",
    "tan": "#d2b48c",
    "teal": "#008080",
    "thistle": "#d8bfd8",
    "tomato": "#ff6347",
    "turquoise": "#40e0d0",
    "violet": "#ee82ee",
    "wheat": "#f5deb3",
    "white": "#ffffff",
    "whitesmoke": "#f5f5f5",
    "yellow": "#ffff00",
    "yellowgreen": "#9acd32",
}

colorlist =[color[1:]  for color in  [
            colormap["chocolate"],
            colormap["yellowgreen"],
            colormap["olive"],
            colormap["darkkhaki"],
            colormap["sienna"],
            colormap["midnightblue"],
            colormap["slateblue"],
            colormap["steelblue"],
            colormap["palegoldenrod"],
            colormap["maroon"],
            colormap["orangered"],
            colormap["thistle"],
            ]]
def SetColor(index,newcolor):
    if newcolor in colormap.keys():
        colorlist[index] = newcolor
    else:
        raise ValueError("unknown color specifier: %s" % newcolor)
    
def GetColor(index):
        return colorlist[index%len(colorlist)]
 

markertypelist = {
        "arrow": "a",
        "cross": "c",
        "diamond": "d",
        "circle": "o",
        "square": "s",
        "vlinex2data": "v",
        "vlineacross": "V",
        "hlineacross": "h",
        "x": "x",           
        }
 
    
def ConvertLabel2UTF8(labellist):
    newlist = []
    for label in labellist:
        if type(label) == type(u'c'):
            label = label.encode("utf-8")
        else:
            label = str(label)
        newlist.append(label)
    return newlist

def CalcBarAndSpacWidth(width, groupsize, legendcount, showlegend = 'r'):
    barspac = 20
    width = width - barspac
    if showlegend in ['r','l','R','L']:
        width = width * 0.8
    barWidth = width / legendcount
    barWidth = barWidth/(groupsize*2) 
    groupspac = barWidth
    return (barWidth,barspac,groupspac)

def CalcTickAndMax(datalist, tickcount = 5):
    maxValue = max(datalist)+1
    if maxValue == 0:
        tempMax = 1
    else:
        exp = math.floor(math.log10(maxValue))
        tempMax = int(math.ceil(maxValue/math.pow(10, exp)) * math.pow(10,exp))
    tickvalue = tempMax / tickcount
    expTick = float(math.floor(math.log10(tickvalue)))
    tickvalue = int(math.ceil(tickvalue / math.pow(10, expTick)) \
                                * math.pow(10, expTick))
        
    maxValue = tickvalue * tickcount
    
    return (maxValue,tickvalue)

def MakeShortLegend(legend, maxlength = 10):
    shortlegend = []
    for label in legend:
        short = label
        if len(label)>maxlength:
            short = label[0:maxlength/2]+str(len(label))+label[-(maxlength/2):]
        shortlegend.append(short)
    return shortlegend

def MakePercentageLegend(data, legend):
    percentLegend = []
    total = sum(data)
    length_data = len(data)
    for i in range(length_data):
        percentLegend.append("%.2f%%"%(data[i]/total*100))
        legend[i] = "%s(%s)" % (legend[i], percentLegend[i])
    return percentLegend,legend

if GOOGLE_CHART_API_PROVIDER == "GChart":
    pass

if GOOGLE_CHART_API_PROVIDER == "pygooglechart":
    
    from pygooglechart import *
    
    def simpleLineChart(legend,data,size=(600,450),showlegend='r', tickcount=8,title='' ):
        if type(title) == type(u'c'):
            title = title.encode('utf-8')
        legend = ConvertLabel2UTF8(legend)
        (max,tick) = CalcTickAndMax(data, tickcount)
               
        chart = XYLineChart(size[0], size[1], y_range=[0, max])
        if len(title): 
            chart.set_title(title)
        chart.set_colours_within_series(['4d89f9']+colorlist)
        chart.set_line_style(1,3)
        
        chart.add_data([0])
        chart.add_data([0])
        chart.add_data(range(1,(len(data)+1),1))
        chart.add_data(data)
        chart.add_data([len(data)+1])
        chart.add_data([0])
        
        
        for i in range(0,len(data)):
            chart.add_marker(1,i,markertypelist['x'],GetColor(i),15)
            chart.add_marker_text('t'+str(data[i]),'000000',1,i,14)
        
        if showlegend not in ['N','n']:
            chart.set_legend(legend)
            chart.set_legend_position(showlegend)
        
        yAxis = range(0, max+1, tick)
        chart.set_axis_labels(Axis.LEFT, yAxis)
        chart.set_grid(0,100/tickcount, 5, 5)
        # X axis labels
        chart.set_axis_labels(Axis.BOTTOM,['']+legend+[''])
        #chart.set_axis_labels(Axis.BOTTOM,data)    
        
        return chart.get_url()
    
    def singleBarChart(legend,data,size=(600,450),showlegend='r', tickcount=8,title='' ):
        if type(title) == type(u'c'):
            title = title.encode('utf-8')
        legend = ConvertLabel2UTF8(legend)
        
        (max,tick) = CalcTickAndMax(data, tickcount)
        (barwidth,barspac,groupspac) = CalcBarAndSpacWidth(size[0],1, len(legend), showlegend)
        
        chart = GroupedVerticalBarChart(size[0], size[1], y_range=[0, max])
        if len(title): 
            chart.set_title(title)
        chart.set_colours_within_series(colorlist)
        
        chart.add_data(data)
        
        #chart.add_data_line('4d89f9',0,2,priority=1)
        for i in range(len(data)):
            chart.add_marker_text('t'+str(data[i]),'000000',0,i,14)
                
        chart.set_bar_width(barwidth)
        chart.set_bar_spacing(barspac)
        chart.set_group_spacing(groupspac)
       
        if showlegend not in ['N','n']:
            chart.set_legend(legend)
            chart.set_legend_position(showlegend)
        
        # Y axis ticks        
        yAxis = range(0, max+1, tick)
        chart.set_axis_labels(Axis.LEFT, yAxis)
        chart.set_grid(0,100/tickcount, 5, 5)
        # X axis labels
        chart.set_axis_labels(Axis.BOTTOM,legend)
        #chart.set_axis_labels(Axis.BOTTOM,data)
        
        return chart.get_url()
    
    def pieChart(legend,data,size=(600,450),showlegend='r', tickcount=8,title='' ):
        if type(title) == type(u'c'):
            title = title.encode('utf-8')
        legend = ConvertLabel2UTF8(legend)
        percent_legend,legend = MakePercentageLegend(data,legend)
        
        (max,tick) = CalcTickAndMax(data, tickcount)
               
        chart = PieChart2D(size[0], size[1])
        if len(title): 
            chart.set_title(title)
        chart.set_colours_within_series(colorlist)
        chart.set_line_style(1,3)
        
        chart.add_data(data)
        chart.set_pie_labels(legend) 
        chart.set_legend(legend)  
        chart.set_legend_position('b') 

        return chart.get_url()
    
    
    
